package socket.homework;
import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;

/**
 * 聊天室服务端
 */
public class Server1 {
    private ServerSocket serverSocket; //总机
    //private List<PrintWriter> allOut = new ArrayList<>(); //设计集合存放所有客户端输出流，用于群发
    //存放所有客户端的输出流（key：客户端的昵称 value：客户端的输出流）
    private Map<String,PrintWriter> allOut = new HashMap<>();
    public Server1(){
        try {
            System.out.println("正启动服务端");
            /*
              创建ServerSocket对象时，会申请一个端口
              如果该端口被其他程序占用，则会抛出异常
                java.net.BindException: Address already in use:
              解决方法:
              1.杀死占用该端口的进程(通常发生在服务端启动两次导致)
              2.更新端口号
             */
            serverSocket = new ServerSocket(8088);
            System.out.println("服务器启动完毕");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 该方法用于启动服务端程序的执行
     */
    public void start(){
        try {
            while(true){
                System.out.println("等待客户端连接...");
                /*
                  ServerSocket提供的重要方法:
                    Socket accept():用于接收客户端的连接，并返回一个Socket实例与连接的客户端交互
                    该方法是一个阻塞方法，调用后程序会"卡住"，直接一个客户端连接为止。
                 */
                Socket socket = serverSocket.accept();
                System.out.println("一个客户端连接了!");

                //启动一个线程来处理与该客户端的交互
                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server1 server = new Server1();
        server.start();
    }

    /**
     * 该线程任务用于负责与指定的客户端交互
     */
    private class ClientHandler implements Runnable{
        private Socket socket;
        private String host; //记录该客户端的IP地址
        private String nickName; //记录客户端昵称
        public ClientHandler(Socket socket){
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress(); //通过socket获取客户端的IP地址
        }

        public void run(){
            PrintWriter pw = null;
            try {
                //通过socket获取输入流用于读取客户端发送过来的消息
                InputStream in = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);

                //通过socket获取输出流用于向客户端发送消息
                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out,StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw,true);

                nickName = br.readLine(); //读取客户端发送过来的昵称
                synchronized(allOut) { //向集合中添加元素需要添加同步机制，保证多个线程不能同时添加
                    allOut.put(nickName,pw); //将客户端的昵称和输出流存入map中
                }
                sendMessageToAll("上线了，当前在线人数:"+allOut.size()); //通知所有客户端，该用户上线了

                String message;
                while((message=br.readLine())!=null){ //读取客户端发送过来的消息
                    if (message.startsWith("@  ")){//如果聊天内容以@开头，则将消息转发给指定的客户端，否则群发消息
                        sendMessageToSome(message);
                    }else{//群发消息
                        sendMessageToAll(message);
                    }
                    sendMessageToAll("说: "+message); //将消息发送给所有人
                }
            } catch (IOException e) { //冷处理
            } finally {
                //处理客户端下线后的操作，共3步:
                synchronized(allOut){ //从集合中删除元素需要同步机制，保证多个线程不能同时删除
                    allOut.remove(nickName); //根据昵称将该客户端的输出流从共享中删除
                }
                sendMessageToAll("下线了，当前在线人数:"+allOut.size()); //2)通过所有客户端，该用户下线了
                try {
                    socket.close(); //3)关闭socket
                } catch (IOException e) {
                }
            }
        }

        /**
         * 将消息发送给所有客户端(群聊)
         * @param message 客户端发送过来的消息
         */
        private void sendMessageToAll(String message){  //小黑[192.168.0.152]上线了，在线人数
            System.out.println(nickName+"["+host+"]"+message); //服务端输出消息
            synchronized(allOut) {
                /*
                  新循环遍历集合就是迭代器，因为迭代器要求遍历过程中不可以用集合的方式来增删元素，
                  因此为了保证遍历正确，防止其它线程增删，这里要和其它线程的增删互斥
                 */
                for(PrintWriter o : allOut.values()){
                    o.println(nickName+"["+host+"]"+message); //群发消息
                }
            }

        }

        /**
         * 将消息发送给指定用户(私聊)
         */
        private void sendMessageToSome(String message){
            //message的样子：@范传奇: 你好！
            if (message.matches("@.+:.+")) {// 正则表达式,验证私聊格式是否符合
                String toNickName = message.substring(1,message.indexOf(":"));
                if (allOut.containsKey(toNickName)){//在线
                    String toMessage = message.substring(message.indexOf(":")+1);//获取私聊内容
                    PrintWriter toPW = allOut.get(toNickName);//根据昵称获取输出流,allOut是map集合，key是昵称，value是输出流
                    toPW.println(nickName+"悄悄对你说: "+toMessage);
                }else{//不在线
                    PrintWriter pw = allOut.get(nickName);
                    pw.println("该用户["+toNickName+"]不在线!");
                }

            }else {// 私聊格式不正确//如果消息格式不正确，通过 allOut 获取当前客户端的输出流 PrintWriter，并向其发送一条提示消息“私聊格式不正确”。
                PrintWriter pw = allOut.get(nickName);//根据昵称获取当前客户端的输出流
                pw.println("私聊格式不正确");
            }
        }


    }
}